Исходный массив:
a = [[1, 4, 5], [2], [3, 4]]
Отсортировать массив по длине подмассивов:
a.sort(key=lambda x: len(x))
Это то же самое, что и:
def f(x):
return len(x)
a.sort(key=f)
Автоматически (без указания ключа) массив сортируется сначала по первому элементу, потом по второму, и т.д.